'''
@author：fc
@date：  2021/11/30
@contact：675435108@qq.com
'''
"""
文件内容&功能简要：
# 这是一个数据处理工具文件
"""

import pandas as pd


class DataProcess:
    def __init__(self):
        pass

    def process(self):
        self.m1_1l_dat_to_csv()

    """
         转换ml-1m中的数据格式为csv
    """
    def m1_1l_dat_to_csv(self):
        print("开始转换用户数据（users.dat）...")
        ml_1m = pd.read_table("../data/ml-1m/users.dat", sep="::", engine='python',
                              names=['userId', 'gender', 'age', 'occupation', 'zip-code'])
        ml_1m.to_csv("../data/generate/ml-1m/users.csv", index=False)
        print("开始转换电影数据（movies.dat）...")
        ml_1m = pd.read_table("../data/ml-1m/movies.dat", sep="::", engine='python',
                              names=['movieId', 'title', 'genres'])
        ml_1m.to_csv("../data/generate/ml-1m/movies.csv", index=False)
        print("开始转换评分数据（ratings.dat）...")
        ml_1m = pd.read_table("../data/ml-1m/ratings.dat", sep="::", engine='python',
                              names=['userId', 'movieId', 'rating', 'timaStamp'])
        ml_1m.to_csv("../data/generate/ml-1m/ratings.csv", index=False)
        print("over")


if __name__ == '__main__':
    dp=DataProcess()
    dp.process()

